Defining a quadratic function with numpy.meshgrid